public class code {
    //给定 x, 把一个链表整理成前半部分小于 x, 后半部分大于等于 x 的形式:分成两部分，然后相连，注意一边没有元素；还有最后一个元素的next不为null
    public ListNode partition(ListNode pHead, int x) {
        if(pHead == null) {
            return null;
        }
        ListNode cur = pHead;
        ListNode bs = null;
        ListNode be = null;
        ListNode as = null;
        ListNode ae = null;
        //开始遍历排序
        while(cur != null) {
            //小于x
            if(cur.val <x) {
                //bs和be还未有元素
                if(bs == null) {
                    bs = cur;
                    be = cur;
                } else{
                    be.next = cur;
                    be = be.next;
                }
                //大于x
            } else {
                //as和ae还有元素
                if(as == null) {
                    as = cur;
                    ae = cur;
                } else {
                    ae.next = cur;
                    ae = ae.next;
                }

            }
            cur = cur.next;
        }
        //连起来
        //没有小于x的值
        if(bs == null) {
            return as;
        }
        be.next = as;
        //最后元素的next要置为null，不然可能为前面的地址
        if(as != null) {//有大于x的元素
            ae.next = null;
        }
        return bs;
    }
}
